package com.study.order.amqp;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

@Slf4j
@Configuration
public class RabbitMqConfig {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 序列化MQ中的内容为JSON方式
     *
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制 RabbitTemplate
     * 1.服务器收到消息就回调
     *  1.1 spring.rabbitmq.publisher-confirm-type=CORRELATED
     *  1.2 设置确认回调 ConfirmCallback
     * 2.消息正确抵达队列进行回调
     *  2.1  spring.rabbitmq.publisher-returns=true
     *  spring.rabbitmq.template.mandatory=true
     *  2.2 设置确认回调 ReturnCallback
     *
     * 3.消费端确认(保证每个消息被正确消费,此时才可以从Broker删除这个消息)
     *   3.1 默认是自动确认的,只要消息接收到,客户端会自动确认,服务端就会移除这个消息
     *      问题:
     *      我们收到很多消息,自动回复给服务器ACK,只有一个消息处理成功,然后宕机了，发生消息丢失
     *      解决:
     *      spring.rabbitmq.listener.simple.acknowledge-mode=manual 开启手动签收
     *      消费者手动确认模式:只要我们没用明确告诉MQ,消息被签收,没有ACK,消息就一直是Unacked状态.
     *         即使 Consumer 宕机,消息不会丢失,会重新变为 Ready,下次有新的 Consumer连接进来就发给他
     *     3.2 如何签收消息
     *
     *     channel.basicAck(deliveryTag,false);签收 业务成功完成
     *     channel.basicNack(deliveryTag,false,false);据签 业务失败拒签
     */
    @PostConstruct //RabbitMqConfig对象创建完成以后,执行这个方法
    public void initRabbitTemplate() {
        //设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 1.只要消息抵达Broker 那么 ack=true
             *
             * @param correlationData 当前消息的唯一关联数据(消息的唯一ID)
             * @param ack             消息时候成功收到
             * @param cause           失败原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                log.info("[correlationData]:{},[ack]:{},[cause]:{}", correlationData, ack, cause);
            }
        });

        //设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列,就触发这个失败回调
             *
             * @param message    投递失败消息的详细信息
             * @param replyCode  回复的状态码
             * @param replyText  回复的文本内容
             * @param exchange   当时这个消息发给哪个交换机
             * @param routingKey 当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                log.info("[message]:{},[replyCode]:{},[replyText]:{},[exchange]:{},[routingKey]:{}", message, replyText, replyText, exchange, routingKey);
            }
        });
    }
}
